from modelscope import snapshot_download
import os

MODELS = {
    "qwen3-embedding-0.6b": "Qwen/Qwen3-Embedding-0.6B",
    "qwen3-reranker-0.6b": "Qwen/Qwen3-Reranker-0.6B",
}

for local_name, hub_name in MODELS.items():
    print(f"Downloading {hub_name} -> models/{local_name}")
    snapshot_download(hub_name, cache_dir=f"models/{local_name}")